home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_mmap.pyo (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-10-18  |  6.8 KB  |  293 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. from test.test_support import verify, vereq, TESTFN
  5. import mmap
  6. import os
  7. import re
  8. PAGESIZE = mmap.PAGESIZE
  9.  
  10. def test_both():
  11.     '''Test mmap module on Unix systems and Windows'''
  12.     if os.path.exists(TESTFN):
  13.         os.unlink(TESTFN)
  14.     
  15.     f = open(TESTFN, 'w+')
  16.     
  17.     try:
  18.         f.write('\x00' * PAGESIZE)
  19.         f.write('foo')
  20.         f.write('\x00' * (PAGESIZE - 3))
  21.         f.flush()
  22.         m = mmap.mmap(f.fileno(), 2 * PAGESIZE)
  23.         f.close()
  24.         print type(m)
  25.         print '  Position of foo:', m.find('foo') / float(PAGESIZE), 'pages'
  26.         vereq(m.find('foo'), PAGESIZE)
  27.         print '  Length of file:', len(m) / float(PAGESIZE), 'pages'
  28.         vereq(len(m), 2 * PAGESIZE)
  29.         print '  Contents of byte 0:', repr(m[0])
  30.         vereq(m[0], '\x00')
  31.         print '  Contents of first 3 bytes:', repr(m[0:3])
  32.         vereq(m[0:3], '\x00\x00\x00')
  33.         print "\n  Modifying file's content..."
  34.         m[0] = '3'
  35.         m[PAGESIZE + 3:PAGESIZE + 3 + 3] = 'bar'
  36.         print '  Contents of byte 0:', repr(m[0])
  37.         vereq(m[0], '3')
  38.         print '  Contents of first 3 bytes:', repr(m[0:3])
  39.         vereq(m[0:3], '3\x00\x00')
  40.         print '  Contents of second page:', repr(m[PAGESIZE - 1:PAGESIZE + 7])
  41.         vereq(m[PAGESIZE - 1:PAGESIZE + 7], '\x00foobar\x00')
  42.         m.flush()
  43.         match = re.search('[A-Za-z]+', m)
  44.         if match is None:
  45.             print '  ERROR: regex match on mmap failed!'
  46.         else:
  47.             (start, end) = match.span(0)
  48.             length = end - start
  49.             print '  Regex match on mmap (page start, length of match):', start / float(PAGESIZE), length
  50.             vereq(start, PAGESIZE)
  51.             vereq(end, PAGESIZE + 6)
  52.         m.seek(0, 0)
  53.         print '  Seek to zeroth byte'
  54.         vereq(m.tell(), 0)
  55.         m.seek(42, 1)
  56.         print '  Seek to 42nd byte'
  57.         vereq(m.tell(), 42)
  58.         m.seek(0, 2)
  59.         print '  Seek to last byte'
  60.         vereq(m.tell(), len(m))
  61.         print '  Try to seek to negative position...'
  62.         
  63.         try:
  64.             m.seek(-1)
  65.         except ValueError:
  66.             pass
  67.  
  68.         verify(0, 'expected a ValueError but did not get it')
  69.         print '  Try to seek beyond end of mmap...'
  70.         
  71.         try:
  72.             m.seek(1, 2)
  73.         except ValueError:
  74.             pass
  75.  
  76.         verify(0, 'expected a ValueError but did not get it')
  77.         print '  Try to seek to negative position...'
  78.         
  79.         try:
  80.             m.seek(-len(m) - 1, 2)
  81.         except ValueError:
  82.             pass
  83.  
  84.         verify(0, 'expected a ValueError but did not get it')
  85.         print '  Attempting resize()'
  86.         
  87.         try:
  88.             m.resize(512)
  89.         except SystemError:
  90.             pass
  91.  
  92.         verify(len(m) == 512, 'len(m) is %d, but expecting 512' % (len(m),))
  93.         
  94.         try:
  95.             m.seek(513, 0)
  96.         except ValueError:
  97.             pass
  98.  
  99.         verify(0, 'Could seek beyond the new size')
  100.         f = open(TESTFN)
  101.         f.seek(0, 2)
  102.         verify(f.tell() == 512, 'Underlying file not truncated')
  103.         f.close()
  104.         verify(m.size() == 512, 'New size not reflected in file')
  105.         m.close()
  106.     finally:
  107.         
  108.         try:
  109.             f.close()
  110.         except OSError:
  111.             pass
  112.  
  113.         
  114.         try:
  115.             os.unlink(TESTFN)
  116.         except OSError:
  117.             pass
  118.  
  119.  
  120.     
  121.     try:
  122.         mapsize = 10
  123.         print '  Creating', mapsize, 'byte test data file.'
  124.         open(TESTFN, 'wb').write('a' * mapsize)
  125.         print '  Opening mmap with access=ACCESS_READ'
  126.         f = open(TESTFN, 'rb')
  127.         m = mmap.mmap(f.fileno(), mapsize, access = mmap.ACCESS_READ)
  128.         verify(m[:] == 'a' * mapsize, 'Readonly memory map data incorrect.')
  129.         print "  Ensuring that readonly mmap can't be slice assigned."
  130.         
  131.         try:
  132.             m[:] = 'b' * mapsize
  133.         except TypeError:
  134.             pass
  135.  
  136.         verify(0, 'Able to write to readonly memory map')
  137.         print "  Ensuring that readonly mmap can't be item assigned."
  138.         
  139.         try:
  140.             m[0] = 'b'
  141.         except TypeError:
  142.             pass
  143.  
  144.         verify(0, 'Able to write to readonly memory map')
  145.         print "  Ensuring that readonly mmap can't be write() to."
  146.         
  147.         try:
  148.             m.seek(0, 0)
  149.             m.write('abc')
  150.         except TypeError:
  151.             pass
  152.  
  153.         verify(0, 'Able to write to readonly memory map')
  154.         print "  Ensuring that readonly mmap can't be write_byte() to."
  155.         
  156.         try:
  157.             m.seek(0, 0)
  158.             m.write_byte('d')
  159.         except TypeError:
  160.             pass
  161.  
  162.         verify(0, 'Able to write to readonly memory map')
  163.         print "  Ensuring that readonly mmap can't be resized."
  164.         
  165.         try:
  166.             m.resize(2 * mapsize)
  167.         except SystemError:
  168.             pass
  169.         except TypeError:
  170.             pass
  171.  
  172.         verify(0, 'Able to resize readonly memory map')
  173.         del m
  174.         del f
  175.         verify(open(TESTFN, 'rb').read() == 'a' * mapsize, 'Readonly memory map data file was modified')
  176.         print '  Opening mmap with size too big'
  177.         import sys
  178.         f = open(TESTFN, 'r+b')
  179.         
  180.         try:
  181.             m = mmap.mmap(f.fileno(), mapsize + 1)
  182.         except ValueError:
  183.             if sys.platform.startswith('win'):
  184.                 verify(0, 'Opening mmap with size+1 should work on Windows.')
  185.             
  186.         except:
  187.             sys.platform.startswith('win')
  188.  
  189.         if not sys.platform.startswith('win'):
  190.             verify(0, 'Opening mmap with size+1 should raise ValueError.')
  191.         
  192.         m.close()
  193.         f.close()
  194.         if sys.platform.startswith('win'):
  195.             f = open(TESTFN, 'r+b')
  196.             f.truncate(mapsize)
  197.             f.close()
  198.         
  199.         print '  Opening mmap with access=ACCESS_WRITE'
  200.         f = open(TESTFN, 'r+b')
  201.         m = mmap.mmap(f.fileno(), mapsize, access = mmap.ACCESS_WRITE)
  202.         print '  Modifying write-through memory map.'
  203.         m[:] = 'c' * mapsize
  204.         verify(m[:] == 'c' * mapsize, 'Write-through memory map memory not updated properly.')
  205.         m.flush()
  206.         m.close()
  207.         f.close()
  208.         f = open(TESTFN, 'rb')
  209.         stuff = f.read()
  210.         f.close()
  211.         verify(stuff == 'c' * mapsize, 'Write-through memory map data file not updated properly.')
  212.         print '  Opening mmap with access=ACCESS_COPY'
  213.         f = open(TESTFN, 'r+b')
  214.         m = mmap.mmap(f.fileno(), mapsize, access = mmap.ACCESS_COPY)
  215.         print '  Modifying copy-on-write memory map.'
  216.         m[:] = 'd' * mapsize
  217.         verify(m[:] == 'd' * mapsize, 'Copy-on-write memory map data not written correctly.')
  218.         m.flush()
  219.         verify(open(TESTFN, 'rb').read() == 'c' * mapsize, 'Copy-on-write test data file should not be modified.')
  220.         
  221.         try:
  222.             print '  Ensuring copy-on-write maps cannot be resized.'
  223.             m.resize(2 * mapsize)
  224.         except TypeError:
  225.             pass
  226.  
  227.         verify(0, 'Copy-on-write mmap resize did not raise exception.')
  228.         del m
  229.         del f
  230.         
  231.         try:
  232.             print '  Ensuring invalid access parameter raises exception.'
  233.             f = open(TESTFN, 'r+b')
  234.             m = mmap.mmap(f.fileno(), mapsize, access = 4)
  235.         except ValueError:
  236.             pass
  237.  
  238.         verify(0, 'Invalid access code should have raised exception.')
  239.         if os.name == 'posix':
  240.             f = open(TESTFN, 'r+b')
  241.             
  242.             try:
  243.                 m = mmap.mmap(f.fileno(), mapsize, flags = mmap.MAP_PRIVATE, prot = mmap.PROT_READ, access = mmap.ACCESS_WRITE)
  244.             except ValueError:
  245.                 pass
  246.  
  247.             verify(0, 'Incompatible parameters should raise ValueError.')
  248.             f.close()
  249.     finally:
  250.         
  251.         try:
  252.             os.unlink(TESTFN)
  253.         except OSError:
  254.             pass
  255.  
  256.  
  257.     f = open(TESTFN, 'w+')
  258.     
  259.     try:
  260.         data = 'aabaac\x00deef\x00\x00aa\x00'
  261.         n = len(data)
  262.         f.write(data)
  263.         f.flush()
  264.         m = mmap.mmap(f.fileno(), n)
  265.         f.close()
  266.         for start in range(n + 1):
  267.             for finish in range(start, n + 1):
  268.                 slice = data[start:finish]
  269.                 vereq(m.find(slice), data.find(slice))
  270.                 vereq(m.find(slice + 'x'), -1)
  271.             
  272.         
  273.         m.close()
  274.     finally:
  275.         os.unlink(TESTFN)
  276.  
  277.     f = open(TESTFN, 'w+')
  278.     
  279.     try:
  280.         f.write(2 ** 16 * 'a')
  281.         f.close()
  282.         f = open(TESTFN)
  283.         mf = mmap.mmap(f.fileno(), 2 ** 16, access = mmap.ACCESS_READ)
  284.         mf.close()
  285.         mf.close()
  286.         f.close()
  287.     finally:
  288.         os.unlink(TESTFN)
  289.  
  290.     print ' Test passed'
  291.  
  292. test_both()
  293.